'''
1.2新增带看量字段
1.3修改了Header的构筑方式,变更字段，文件带时间
'''
import os
import time
import pandas as pd
#获取文件目录
file_path=os.getcwd()
#构建文件地址
STORE_PATH=file_path+'\\门店数据\\'
AREA_PATH=file_path+'\\区域数据\\'
SAVE_PATH=file_path+'\\对比结果\\'
def DATA_CP():
    global S_result,A_result,DF_MERGE,DF_MERGE_A,i
    STORE_PATH_DIR=os.listdir(STORE_PATH)
    AREA_PATH_DIR=os.listdir(AREA_PATH)
    #读取文件,index_col=0是将第一列变成DataFrame里的行索引
    #门店
    DF_NOW=pd.read_excel(STORE_PATH+STORE_PATH_DIR[1],header=0,index_col=0)
    DF_LAST=pd.read_excel(STORE_PATH+STORE_PATH_DIR[0],header=0,index_col=0)
    #区域
    DF_NOW_A=pd.read_excel(AREA_PATH+AREA_PATH_DIR[1],header=0,index_col=0)
    DF_LAST_A=pd.read_excel(AREA_PATH+AREA_PATH_DIR[0],header=0,index_col=0)
    #构建表头
    Header_LIST=['经纪人人数','新增房源-二手新增','二手房人均新增',\
                 '新增房源-租赁新增','租赁人均新增','二手客户-新增',\
                 '二手房客户人均新增','租赁客户-新增','租赁客户人均新增',\
                 '二手新增带看','二手人均新增带看','租赁新增带看',\
                 '租赁人均新增带看','二手房总数-公盘','二手房人均公盘',\
                 '二手房总数-私盘','二手房人均私盘','租赁总数-公盘',\
                 '租赁人均公盘','租赁总数-私盘','租赁人均私盘']
    #取文件交集，并将行列转置
    DF_MERGE=pd.merge(DF_NOW,DF_LAST,how='left',left_index=True,right_index=True)
    DF_MERGE_A=pd.merge(DF_NOW_A,DF_LAST_A,how='left',left_index=True,right_index=True)
    #消去NaN值为0
    DF_MERGE=DF_MERGE.fillna(0.00)
    DF_MERGE_A=DF_MERGE_A.fillna(0.00)
    #门店数据
    S_result=pd.DataFrame()
    #区域数据
    A_result=pd.DataFrame()
    ##-----------------------------------------------
    #门店数据
    for i in Header_LIST:
        S_result[i]=(DF_MERGE[i+'_x']-DF_MERGE[i+'_y'])
    S_result.sort_values(by="新增房源-二手新增", inplace=True, ascending=True)
    #区域数据
    for i in Header_LIST:
        A_result[i]=(DF_MERGE_A[i+'_x']-DF_MERGE_A[i+'_y'])
    A_result.sort_values(by="新增房源-二手新增", inplace=True, ascending=True)
#时间
Str_time=time.strftime("%Y-%m-%d",time.localtime())
#执行第一次程序
DATA_CP()
S_result.to_excel(SAVE_PATH+Str_time+'门店数据对比结果.xlsx')
A_result.to_excel(SAVE_PATH+Str_time+'区域数据对比结果.xlsx')
#修改目录执行第二次用来筛选综合大区
STORE_PATH=file_path+'\\综合大区门店数据\\'
AREA_PATH=file_path+'\\综合大区区域数据\\'
DATA_CP()
S_result.to_excel(SAVE_PATH+Str_time+'综合门店结果.xlsx')
A_result.to_excel(SAVE_PATH+Str_time+'综合区域结果.xlsx')
print('完成')
